package com.subsequent.utils;


import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.Random;
import java.util.concurrent.TimeUnit;

/**
 * 随机生成验证码工具类
 */
@Component
public class ValidateCodeUtils {

    private final RedisTemplate<String, String> redisTemplate;

    // 验证码有效期（分钟）
    private static final long CODE_EXPIRE_MINUTES = 5;

    // 验证码发送间隔（秒）
    private static final long CODE_SEND_INTERVAL = 60;

    // Redis key前缀
    private static final String CODE_PREFIX = "validate_code:";
    private static final String SEND_COUNT_PREFIX = "send_count:";

    public ValidateCodeUtils(RedisTemplate<String, String> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    /**
     * 随机生成验证码
     * @param length 长度为4位或者6位
     * @return
     */
    public static Integer generateValidateCode(int length){
        Integer code = null;
        if(length == 4){
            code = new Random().nextInt(10000);//生成随机数，最大为9999
            if(code < 1000){
                code = code + 1000;//保证随机数为4位数字
            }
        }else if(length == 6){
            code = new Random().nextInt(999999);//生成随机数，最大为999999
            if(code < 100000){
                code = code + 100000;//保证随机数为6位数字
            }
        }else{
            throw new RuntimeException("只能生成4位或6位数字验证码");
        }
        return code;
    }

    /**
     * 保存验证码到Redis
     * @param phone 手机号
     * @param code 验证码
     * @return 是否保存成功
     */
    public boolean saveValidateCode(String phone, String code) {
        if (!StringUtils.hasText(phone) || !StringUtils.hasText(code)) {
            return false;
        }

        String key = CODE_PREFIX + phone;
        try {
            redisTemplate.opsForValue().set(key, code, CODE_EXPIRE_MINUTES, TimeUnit.MINUTES);
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 验证验证码
     * @param phone 手机号
     * @param code 验证码
     * @return 是否验证通过
     */
    public boolean validateCode(String phone, String code) {
        if (!StringUtils.hasText(phone) || !StringUtils.hasText(code)) {
            return false;
        }

        String key = CODE_PREFIX + phone;
        try {
            String savedCode = redisTemplate.opsForValue().get(key);
            if (savedCode != null && savedCode.equals(code)) {
                // 验证通过后删除验证码
                redisTemplate.delete(key);
                return true;
            }
            return false;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 检查是否可以发送验证码
     * @param phone 手机号
     * @return 是否可以发送
     */
    public boolean canSendCode(String phone) {
        if (!StringUtils.hasText(phone)) {
            return false;
        }

        String key = SEND_COUNT_PREFIX + phone;
        try {
            String count = redisTemplate.opsForValue().get(key);
            if (count == null) {
                // 第一次发送，设置计数器
                redisTemplate.opsForValue().set(key, "1", CODE_SEND_INTERVAL, TimeUnit.SECONDS);
                return true;
            }

            int sendCount = Integer.parseInt(count);
            if (sendCount >= 5) { // 限制5分钟内最多发送5次
                return false;
            }

            // 增加发送次数
            redisTemplate.opsForValue().increment(key);
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    /**
     * 获取验证码剩余有效期（秒）
     * @param phone 手机号
     * @return 剩余有效期，-1表示不存在
     */
    public long getCodeExpireTime(String phone) {
        if (!StringUtils.hasText(phone)) {
            return -1;
        }

        String key = CODE_PREFIX + phone;
        try {
            Long ttl = redisTemplate.getExpire(key);
            return ttl != null ? ttl : -1;
        } catch (Exception e) {
            return -1;
        }
    }
}